"""
python 2d_bird_pyswarm.py
"""

from pyswarm import pso

# 定义目标函数
def objective_function(xy):
    x, y = xy  # 提取x和y
    return x**2 + y**2

# 定义搜索空间
lower_bounds = [-10, -10]  # x和y的下界
upper_bounds = [10, 10]    # x和y的上界

# 调用PSO算法进行优化
best_position, best_value = pso(objective_function, lower_bounds, upper_bounds, swarmsize=5, maxiter=10000)

# 输出结果
print("最优位置 (x, y):", best_position)
print("最小值 f(x, y):", best_value)
